# Requires a CUDA-enabled PyTorch (see README for install commands).
# CPU-only builds will fail because the scripts use .to("cuda") internally.
torch==2.5.1
torchvision==0.20.1
torchaudio==2.5.1
numpy>=1.24
pandas>=2.0
PyYAML>=6.0
tqdm>=4.60
scikit-learn>=1.3 
